import pandas as pd

from cement_prediction_model_microservice.new_model_server.catboost_model import model_architecture, predict_data, training_data

def test_cm_train(train_data: pd.DataFrame):
    """
    测试Catboost模型
    :param traning_data:
    :return:
    """
    train_data = train_data.copy()
    model = model_architecture.NonLinearModel(train_data, model_name='catboost', save_model_path='./model')
    model.train_model_3d()


